import argparse


def parse_args():
    parser = argparse.ArgumentParser(description="FM")

    parser.add_argument('--epoch', type=int, default=100, help='number of epochs')
    parser.add_argument('--run_id', type=int, default=1)
    parser.add_argument('--stop_window', type=int, default=5)
    parser.add_argument('--dim', type=int, default=64, help='embedding size')
    parser.add_argument('--l2', type=float, default=0.002, help='l2 regularization weight')
    parser.add_argument('--lr', type=float, default=2, help='learning rate')
    parser.add_argument('--init', type=float, default=0.66, help='模型的初始化参数')
    parser.add_argument('--model', type=str, default='fm', help='fm,lr,ffm')
    parser.add_argument('--pre_train', type=str, default='no', help='fm,lr,ffm')
    parser.add_argument('--model_file', type=str, default='no', help='fm,lr,ffm')



    return parser.parse_args()
